python prime number code